package com.liuhuan.interceptor;

import cn.hutool.json.JSONUtil;
import com.liuhuan.api.CommonResult;
import com.liuhuan.utils.JwtUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @version 1.0
 * @Author liuhuan
 * @Date 2024/7/4 19:48
 * @注释
 */


@Slf4j
@Component
public class LoginCheckInterceptor implements HandlerInterceptor {
    @Override //true: 放行, false, 不放行
    public boolean preHandle(HttpServletRequest req, HttpServletResponse resp, Object handler) throws Exception {
        //1.获取请求url。
        String url = req.getRequestURL().toString();
        System.out.println(url);
        //2.登录操作放行
        if(url.contains("login")){
            log.info("用户登录");
            return true;
        }


        //3.获取请求头中的令牌（token）。
        String jwt = req.getHeader("token");

        //4.判断令牌是否存在，如果不存在，返回错误结果（未登录）。
        if(!StringUtils.hasLength(jwt)){
            log.info("token为空");
            CommonResult error = CommonResult.fail("NOT_LOGIN");
            //手动转换 对象--json
            String notLogin = JSONUtil.toJsonStr(error);
            resp.getWriter().write(notLogin);
            return false;
        }

        //5.解析token，如果解析失败，返回错误结果（未登录）。
        try {
            JwtUtils.parseJWT(jwt);
        } catch (Exception e) {//jwt解析失败
            e.printStackTrace();
            log.info("解析令牌失败");
            CommonResult error = CommonResult.fail("NOT_LOGIN");
            //手动转换 对象--json -
            String notLogin = JSONUtil.toJsonStr(error);
            resp.getWriter().write(notLogin);
            return false;
        }

        //6.放行。
        log.info("令牌合法, 放行");
        return true;
    }

    @Override //目标资源方法运行后运行
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

    }

    @Override //视图渲染完毕后运行, 最后运行
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {

    }
}
